package com.xu.ai.model.openai.enums;

import org.springframework.ai.model.ChatModelDescription;

/**
 * 大模型名称枚举类
 *
 * @author xuguan
 * @since 2025/9/24
 */
public enum ChatModelEnum implements ChatModelDescription {
    QWEN_PLUS("qwen-plus"),
    QWEN_TURBO("qwen-turbo"),
    QWEN_MAX("qwen-max"),
    QWEN_MAX_LONGCONTEXT("qwen-max-longcontext"),
    QWQ_PLUS("qwq-plus"),
    QWQ_PLUS_LATEST("qwen-plus-latest"),
    QWEN_3_32B("qwq-32b"),
    QWEN_OMNI_TURBO("qwen-omni-turbo"),
    QWEN_VL_MAX("qwen-vl-max"),
    DEEPSEEK_R1("deepseek-r1"),
    DEEPSEEK_V3("deepseek-v3"),
    DEEPSEEK_V3_1("deepseek-v3.1"),
    ;

    private final String name;

    ChatModelEnum(String modelName) {
        this.name = modelName;
    }

    @Override
    public String getName() {
        return name;
    }
}
